def argmax():
    pass
